package com.mfangsoft.zhuangjialong.common.webscoket.handler;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.CopyOnWriteArraySet;

import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;

/**
 * @ServerEndpoint 注解是一个类层次的注解，它的功能主要是将目前的类定义成一个websocket服务器端,
 * 注解的值将被用于监听用户连接的终端访问URL地址,客户端可以通过这个URL来连接到WebSocket服务器端
 */
@ServerEndpoint(value = "/app/websocket/{param}")
public class WebSocketTest {

	//concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。若要实现服务端与单一客户端通信的话，可以使用Map来存放，其中Key可以为用户标识
//	private static CopyOnWriteArraySet<WebSocketTest> webSocketSet = new CopyOnWriteArraySet<WebSocketTest>();

	public static Map<String,WebSocketTest> webSocketSet=new HashMap<String,WebSocketTest>();
	//与某个客户端的连接会话，需要通过它来给客户端发送数据
	private Session session;

	/**
	 * 连接建立成功调用的方法
	 * @param session  可选的参数。session为与某个客户端的连接会话，需要通过它来给客户端发送数据
	 * @throws IOException 
	 */
	@OnOpen
	public void onOpen(@PathParam(value="param") String param,Session session) throws IOException{
		System.out.println("传入参数为"+param);
		this.session = session;
		WebSocketTest socketTest = webSocketSet.get(param);
		if(socketTest!=null){
			socketTest.sendMessage("exit");
		}
		webSocketSet.put(param, this);     //加入set中
		System.out.println("有新连接加入！");
	}

	/**
	 * 连接关闭调用的方法
	 * @throws IOException 
	 */
	@OnClose
	public void onClose() throws IOException{
		System.out.println("有一连接关闭！");
	}

	/**
	 * 收到客户端消息后调用的方法
	 * @param message 客户端发送过来的消息
	 * @param session 可选的参数
	 * @throws IOException 
	 */
	@OnMessage
	public void onMessage(@PathParam(value="param") String param,String message, Session session) throws IOException {
		System.out.println(message);
		if("exit".equals(message)){
			webSocketSet.get(param).session.close();
			webSocketSet.remove(param);
		}
		if("hello".equals(message)){
			webSocketSet.get(param).sendMessage("world");
		}
	}

	/**
	 * 发生错误时调用
	 * @param session
	 * @param error
	 */
	@OnError
	public void onError(@PathParam(value="param") String param,Session session, Throwable error){
		System.out.println("发生错误");
		webSocketSet.remove(param);
	}

	/**
	 * 这个方法与上面几个方法不一样。没有用注解，是根据自己需要添加的方法。
	 * @param message
	 * @throws IOException
	 */
	public void sendMessage(String message) throws IOException{
		this.session.getBasicRemote().sendText(message);
	}
}
